package unionfindset;

import java.util.Arrays;

public class UnionFindSet {
    public int[] elem;
    public int usedSize;
    public UnionFindSet(int n) {
        elem=new int[n];
        Arrays.fill(elem,-1);
    }

    /**
     * 找到集合的根节点
     * @param x
     * @return
     */
    public int FindRoot(int x) {
        while (elem[x] >= 0) {
            x=elem[x];
        }
        return x;
    }

    /**
     * 检查这两个数据是否在同一个集合
     * @param x1
     * @param x2
     * @return
     */
    public boolean isSameUnionFindSet(int x1,int x2) {
        int index1=FindRoot(x1),index2=FindRoot(x2);
        if(index1==index2) return true;
        return false;
    }

    /**
     * 合并两个集合
     * @param x1
     * @param x2
     */
    public void union(int x1,int x2) {
        int index1=FindRoot(x1),index2=FindRoot(x2);
        if(index1==index2) return;
        elem[index1]=elem[index1]+elem[index2];
        elem[index2]=index1;
    }

    /**
     * 获取集合数量
     */
    public int getCount() {
        int count=0;
        for (int x : elem) {
            if(x<0) {
                count++;
            }
        }
        return count;
    }
}
